package com.itheima.reggie.service.impl;

import com.baomidou.mybatisplus.core.conditions.query.LambdaQueryWrapper;
import com.baomidou.mybatisplus.extension.service.impl.ServiceImpl;
import com.itheima.reggie.common.CustomException;
import com.itheima.reggie.entity.Category;
import com.itheima.reggie.entity.Dish;
import com.itheima.reggie.entity.Setmeal;
import com.itheima.reggie.mapper.CategoryMapper;
import com.itheima.reggie.service.CategoryService;
import com.itheima.reggie.service.DishService;
import com.itheima.reggie.service.SetmealService;
import jakarta.annotation.Resource;
import org.springframework.stereotype.Service;

import java.io.Serializable;

/**
 * @author GRS
 * @since 2024/4/19 下午5:02
 */
@Service
public class CategoryServiceImpl extends ServiceImpl<CategoryMapper, Category> implements CategoryService
{
    @Resource
    private DishService dishService;

    @Resource
    private SetmealService setmealService;

    @Override
    public boolean removeById(Serializable id)
    {
        LambdaQueryWrapper<Dish> dishWrapper = new LambdaQueryWrapper<>();
        // 添加查询条件，根据分类ID进行查询
        dishWrapper.eq(Dish::getCategoryId, id);
        long dishCount = dishService.count(dishWrapper);

        // 查询当前分类是否关联了菜品，如果已经关联，抛出一个业务异常
        if (dishCount > 0)
            throw new CustomException("当前分类下关联了菜品，不能删除");

        // 查询当前分类是否关联了套餐，如果已经关联，抛出一个业务异常
        LambdaQueryWrapper<Setmeal> setmealWrapper = new LambdaQueryWrapper<>();
        setmealWrapper.eq(Setmeal::getCategoryId, id);
        long setmealCount = setmealService.count(setmealWrapper);

        if (setmealCount > 0)
            throw new CustomException("当前分类下关联了套餐，不能删除");

        // 正常删除分类
        return super.removeById(id);
    }
}